package com.example.mq.mqserver.core;

import com.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

//MSG = Message
/*
* 这个类表示一个消息队列
* */

public class MSGQueue {
    //表示队列的身份标识
    private String name;
    //表示队列是否持久化，ture表示持久化保存，，false表示不持久化保存
    private boolean durable;
    //表示是否可以被一个消费者使用(别人不能用) 这个功能在后续也没有实现
    private boolean exclusive = false;
    //为true 表示没有人使用后，就自动删除,这个功能在后续也没实现
    private boolean autoDelete = false;
    //这个表示扩展参数，也是暂时没有实现的功能
    private Map<String, Object> arguments = new HashMap<>();
    // 当前队列都有哪些消费者订阅了
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();

    // 记录当前取到了第几个消费者，方便实现轮询策略  此处考虑到线程安全问题就不用基本数据类型 int ，用下面的AtomicInteger
    // 就可以保证线程安全
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    // 添加一个新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    // 订阅者的删除暂时先不考虑
    // 挑选一个订阅者，用来处理当前的消息，（按照轮询的方式）
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.size() == 0) {
            // 该队列时没有人订阅的
            return null;
        }
        // 计算一下当前要取的元素的下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement(); // 相当于 ++ 操作
        return consumerEnvList.get(index);
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public boolean isDurable() {
        return durable;
    }

    public void setDurable(boolean durable) {
        this.durable = durable;
    }

    public boolean isExclusive() {
        return exclusive;
    }

    public void setExclusive(boolean exclusive) {
        this.exclusive = exclusive;
    }

    public boolean isAutoDelete() {
        return autoDelete;
    }

    public void setAutoDelete(boolean autoDelete) {
        this.autoDelete = autoDelete;
    }

    public String getArguments() {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            //将 arguments 转换成 json 格式的字符串
           return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            //这个代码的意思是 Mybatis 从数据库中读数据（读的是字符串的格式）之后自动调用 set 方法
            //然后将读到的数据设置进来，然后通过 set 方法将字符串的格式转换成 HashMap 的类型
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String, Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }
}
